#pragma once
#include "Heap.h"

// 堆相关函数（保留原有代码）
//初始化
void HeapInit(hp* php)
{
	assert(php != nullptr);

	//初始化4空间
	php->arr = (DataType*)malloc(sizeof(DataType) * 4);

	if (php->arr == nullptr)
	{
		perror("malloc fail");
		return;
	}

	//实际存储个数
	php->size = 0;

	//可存储最大个数
	php->capacity = 4;
}

//向上调整
void UpAdjust(DataType* a, int child)
{
	int parent = (child - 1) / 2;
	while (child > 0)
	{
		if (a[child] > a[parent])
		{
			std::swap(a[child], a[parent]);
			child = parent;
			parent = (child - 1) / 2;
		}
		else break;
	}
}

//插入数据[向上调整]
void HeapPush(hp* php, DataType x)
{
	assert(php != nullptr);

	//扩容
	if (php->size == php->capacity)
	{
		DataType* tmp = (DataType*)realloc(php->arr, sizeof(DataType) * php->capacity * 2);
		if (tmp == nullptr)
		{
			perror("realloc fail");
			return;
		}
		php->arr = tmp;
		//成功扩容后更新容量
		php->capacity *= 2;
	}
	php->arr[php->size++] = x;

	UpAdjust(php->arr, php->size - 1);
}

//向下调整
void DownAdjust(DataType* a, int n, int parent)
{
	//找最小孩子
	int child = parent * 2 + 1;
	while (child < n)
	{
		if (child + 1 < n && a[child + 1] < a[child])
			++child;

		if (a[child] < a[parent])
		{
			std::swap(a[parent], a[child]);
			parent = child;
			child = parent * 2 + 1;
		}
		else break;
	}
}

//删除堆顶元素
void HeapPop(hp* php)
{
	assert(php != nullptr);
	assert(!HeapEmpty(php));
	std::swap(php->arr[0], php->arr[php->size - 1]);
	(php->size)--;
	DownAdjust(php->arr, php->size, 0);
}

//取堆顶元素
DataType HeapTop(hp* php)
{
	assert(php != nullptr);
	return php->arr[0];
}
//堆是否为空
bool HeapEmpty(hp* php)
{
	assert(php != nullptr);
	return php->size == 0;
}
//元素个数
int HeapSize(hp* php)
{
	assert(php != nullptr);
	return php->size;
}

//销毁堆
void HeapDestory(hp* php)
{
	assert(php != nullptr);
	free(php->arr);
	php->arr = nullptr;
	php->capacity = php->size = 0;
}
